// Copyright (C) Kumo inc. and its affiliates.
// Author: Jeff.li lijippy@163.com
// All rights reserved.
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero General Public License as published
// by the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU Affero General Public License for more details.
//
// You should have received a copy of the GNU Affero General Public License
// along with this program.  If not, see <https://www.gnu.org/licenses/>.
//

#pragma once

#include <cantor/parser/parser.h>
#include <cantor/proto/parser/ast.pb.h>
#include <turbo/utility/status.h>
#include <cantor/transform/transform_base.h>

namespace cantor::transform {
    struct TransExpr : public TransformBase {
        ~TransExpr() override = default;

        void resolve_ast(const cantor::parser::DispatchExpr *node);

        const turbo::Status &check() override;

        cantor::pb::ast::DispatchExpr expr;

        static turbo::Status resolve_to(turbo::Nonnull<const cantor::parser::DispatchExpr *> node,
                                        turbo::Nonnull<cantor::pb::ast::DispatchExpr *> proto_node);

    private:
        void resolve_constant_expr(const cantor::parser::DispatchExpr *node);

        void resolve_range_var(const cantor::parser::DispatchExpr *node);

        void resolve_row_expr(const cantor::parser::DispatchExpr *node);

        /// ET_DQL
        void resolve_subquery_expr(const cantor::parser::DispatchExpr *node);

        void resolve_exists_subquery_expr(const cantor::parser::DispatchExpr *node);

        void resolve_common_subquery_expr(const cantor::parser::DispatchExpr *node);

        void resolve_compare_subquery_expr(const cantor::parser::DispatchExpr *node);

        /// func
        void resolve_func_expr(const cantor::parser::DispatchExpr *node);
        void resolve_ctor_map_func(const cantor::parser::DispatchExpr *node);
        void resolve_ctor_struct_func(const cantor::parser::DispatchExpr *node);
        void resolve_ctor_union_func(const cantor::parser::DispatchExpr *node);
        void resolve_ctor_array_func(const cantor::parser::DispatchExpr *node);
        void resolve_ctor_set_func(const cantor::parser::DispatchExpr *node);
        void resolve_ctor_tuple_func(const cantor::parser::DispatchExpr *node);
        void resolve_ctor_json_func(const cantor::parser::DispatchExpr *node);
        void resolve_ctor_json_array_func(const cantor::parser::DispatchExpr *node);
        void resolve_ctor_proto_func(const cantor::parser::DispatchExpr *node);
        void resolve_ctor_enum_func(const cantor::parser::DispatchExpr *node);
        void resolve_ctor_udf_func(const cantor::parser::DispatchExpr *node);
        void resolve_when_clause_func(const cantor::parser::DispatchExpr *node);
        void resolve_case_when_clause_func(const cantor::parser::DispatchExpr *node);
    };
} // namespace cantor::transform
